import openpyxl

from config.config import *


def read_excel():
    # 1.打开 excel 文件
    # 最好使用当前根目录的相对路径
    workbook = openpyxl.load_workbook(EXCEL_FILE)

    # 2.选择 sheet 表
    worksheet = workbook[SHEET_NAME]

    # 3.操作
    data = [] # 空列表, 用于组装字典里面的参数, 因为最后传入参数是需要传列表类型的

    keys = [cell.value for cell in worksheet[2]]

    for row in worksheet.iter_rows(min_row=3, values_only=True):
        dict_data = dict(zip(keys, row))

        # 如果字段值为 True 才会 append 到 data 之后
        if dict_data["is_true"]:
            data.append(dict_data)
        # 如果在之后判断, 就会浪费计算机资源, 因为 False 的数据也添加进去并当作结果返回了

    # 4.关闭 excel 文件
    workbook.close()

    # 5.返回结果
    return data